Multiple inputs in single line python code